from sqlalchemy import create_engine, text
from app.database import SQLALCHEMY_DATABASE_URL
from app import models

# 创建数据库引擎
engine = create_engine(SQLALCHEMY_DATABASE_URL)

def migrate():
    # 删除现有表（如果存在）
    models.Base.metadata.drop_all(bind=engine)
    
    # 创建新表
    models.Base.metadata.create_all(bind=engine)
    
    print("数据库迁移完成！")

if __name__ == "__main__":
    migrate() 